from .models import DiscreteVAE